package com.example.model.tencent.utils;

import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.tms.v20201229.TmsClient;
import com.tencentcloudapi.tms.v20201229.models.TextModerationRequest;
import com.tencentcloudapi.tms.v20201229.models.TextModerationResponse;

import java.nio.charset.StandardCharsets;
import java.util.Base64;

public class TextModerationUtil {

    public static String secretId = "AKIDqY7hIg1OyTlz1rV6dy63mJemCaoqfGdp";
    public static String secretKey = "5xqfHi4HPP7vigSl9NL4WkiTzGI4se9Z";
    public static TextModerationResponse uploadText(String input) {
        TextModerationResponse resp = null;
        try {
            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
            Credential cred = new Credential(secretId,secretKey );
            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("tms.tencentcloudapi.com");
            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);
            // 实例化要请求产品的client对象,clientProfile是可选的
            TmsClient client = new TmsClient(cred, "ap-guangzhou", clientProfile);
            // 实例化一个请求对象,每个接口都会对应一个request对象
            TextModerationRequest req = new TextModerationRequest();
//            String input = "你他妈就是一个大傻逼";
//            byte[] utf8Bytes = input.getBytes(StandardCharsets.UTF_8);
//            String utf8Encoded = Base64.getEncoder().encodeToString(utf8Bytes);
//            System.out.println("UTF-8 编码后: " + utf8Encoded);

            String base64Encoded = Base64.getEncoder().encodeToString(input.getBytes());
//            System.out.println("Base64 编码后: " + base64Encoded);
            req.setContent(base64Encoded);
            req.setBizType("default");
            // 返回的resp是一个TextModerationResponse的实例，与请求对象对应
            resp = client.TextModeration(req);
            // 输出json格式的字符串回包
//            System.out.println(TextModerationResponse.toJsonString(resp));
        } catch (TencentCloudSDKException e) {
            System.out.println(e.toString());
        }

        return resp;
    }
}
